#!/bin/bash
# 设置根目录路径
root_dir="/media/amd/data/long/mdd_3d_constr"

# 设置输出 CSV 文件路径
output_file="result.csv"

# 清空或创建 CSV 文件，并写入表头
echo "Directory,FileCount" > "$output_file"

# 遍历根目录下以 PAT 开头的所有子目录
for dir in "$root_dir"/PAT*; do
  # 判断是否为目录
  if [ -d "$dir" ]; then
    # 检查该目录下是否有 label 目录
    label_dir="$dir/label"
    if [ -d "$label_dir" ]; then
      # 统计 label 目录下的文件数量
      file_count=$(find "$label_dir" -type f | wc -l)
      
      # 如果文件数量大于 65，记录目录名和文件数量到 CSV 文件
      if [ "$file_count" -gt 65 ]; then
        # 提取目录的名称 (仅保留目录名称，不包含路径)
        dir_name=$(basename "$dir")
        echo "$dir_name,$file_count" >> "$output_file"
      fi
    fi
  fi
done

echo "Script executed successfully. Results are saved in $output_file."
